In [3]:
import os
import numpy as np
import matplotlib.pyplot as plt
# Set the number of rows and columns for the subplot grid
rows = 17
cols = 6
# Create a subplot grid
fig, ax = plt.subplots(rows, cols, figsize=(25, 25))
fig.suptitle("Showing one random image from each class", y=1.05, fontsize=24)
# Define the directory where your dataset is located
data_dir = r"C:\Users\hp\Desktop\FOOD\food_data"
# Get the list of food classes (folders) in your dataset
foods_sorted = sorted(os.listdir(data_dir))
# Initialize a variable to keep track of the current food class
food_id = 0
# Iterate over the subplot grid
for i in range(rows):
for j in range(cols):
try:
# Get the name of the current food class
food_selected = foods_sorted[food_id]
food_id += 1
except:
break
# Skip any system files (like .DS_Store on macOS)
if food_selected.startswith('.'):
continue
# Get the list of images in the current food class
food_selected_images = os.listdir(os.path.join(data_dir, food_selected))
# Choose a random image from the current food class
food_selected_random = np.random.choice(food_selected_images)
# Load and display the image
img = plt.imread(os.path.join(data_dir, food_selected, food_selected_random))
ax[i][j].imshow(img)
ax[i][j].set_title(food_selected, pad=10)
# Remove the axis ticks
plt.setp(ax, xticks=[], yticks=[])
# Adjust the layout to prevent overlapping titles
plt.tight_layout()
# Show the plot
plt.show()
